package com.lagou.edu.factory;

import com.lagou.edu.annotation.Autowired;
import com.lagou.edu.annotation.Service;
import com.lagou.edu.annotation.Transactional;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.reflections.Reflections;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.*;

/**
 * @author 应癫
 *
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    private static Map<String,Object> map = new HashMap<>();  // 存储对象

    private static void parseXml(){
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
        // 解析xml
        SAXReader saxReader = new SAXReader();
        try {
            Document document = saxReader.read(resourceAsStream);
            Element rootElement = document.getRootElement();
            List<Element> beanList = rootElement.selectNodes("//bean");
            for (int i = 0; i < beanList.size(); i++) {
                Element element =  beanList.get(i);
                // 处理每个bean元素，获取到该元素的id 和 class 属性
                String id = element.attributeValue("id");        // accountDao
                String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
                // 通过反射技术实例化对象
                Class<?> aClass = Class.forName(clazz);
                Object o = aClass.newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(id,o);

            }

            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
            // 有property子元素的bean就有传值需求
            List<Element> propertyList = rootElement.selectNodes("//property");
            // 解析property，获取父元素
            for (int i = 0; i < propertyList.size(); i++) {
                Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
                String name = element.attributeValue("name");
                String ref = element.attributeValue("ref");

                // 找到当前需要被处理依赖关系的bean
                Element parent = element.getParent();

                // 调用父元素对象的反射功能
                String parentId = parent.attributeValue("id");
                Object parentObject = map.get(parentId);
                // 遍历父对象中的所有方法，找到"set" + name
                Method[] methods = parentObject.getClass().getMethods();
                for (int j = 0; j < methods.length; j++) {
                    Method method = methods[j];
                    if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
                        method.invoke(parentObject,map.get(ref));
                    }
                }

                // 把处理之后的parentObject重新放到map中
                map.put(parentId,parentObject);

            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static void parseAnno(String basePackageName){
        //扫描指定的基础包，并获得所有带有@Service注解的class
        Reflections reflections = new Reflections(basePackageName);
        Set<Class<?>> serviceClz = reflections.getTypesAnnotatedWith(Service.class);
        try{
            //遍历解析带有@Service的class，并创建实例，以service的value为key存入map中
            for (Class clz: serviceClz) {
                Object bean = clz.newInstance();
                Service annotation = (Service) clz.getAnnotation(Service.class);
                map.put(annotation.value(),bean);
            }
        }catch (Exception e){
            e.printStackTrace();
        }
    }

    private static void handleAutowiredAndTransactionl(){
        try{
            ProxyFactory proxyFactory = (ProxyFactory) map.get("proxyFactory");
            //遍历整个map对象
            for(Map.Entry<String,Object> entry : map.entrySet()){
                Object instace = entry.getValue();
                Class<?> instaceClz = instace.getClass();
                //获得当前class中所有的字段，并为带有@Autowire注解的属性注入实例
                Field[] fields = instaceClz.getDeclaredFields();
                for (Field f: fields ) {
                    Autowired autowired = f.getAnnotation(Autowired.class);
                    if (autowired!=null){
                        String value = autowired.value();
                        //从map中取出实例，反射进行注入
                        Object fieldInstance = map.get(value);
                        f.setAccessible(true);
                        f.set(instace,fieldInstance);
                    }
                }

                //*********************  强化Transactional  ***************************

                //判断当前实例是否需要使用代理类
                boolean needProxy = false;
                //并收集需要被增强的方法
                List<String> methodNameList = new ArrayList<>();
                Method[] methods = instaceClz.getMethods();
                for (Method m: methods ) {
                    Transactional transaction = m.getAnnotation(Transactional.class);
                    if (transaction!=null){
                        needProxy = true;
                        methodNameList.add(m.getName());
                    }
                }
                //需要被增强，进一步通过判断是否有实现接口选择代理的方式
                if (needProxy){
                    String beanName = entry.getKey();
                    Object proxy = null;
                    Class<?>[] interfaces = instaceClz.getInterfaces();
                    if (interfaces==null || interfaces.length==0){
                        proxy = proxyFactory.getCglibProxy(map.get(beanName),methodNameList) ;
                    }else{
                        proxy = proxyFactory.getJdkProxy(map.get(beanName),methodNameList) ;
                    }
                    map.put(beanName,proxy);//更新对象为代理类
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }
    }

    static {
        //解析Xml定义的Bean，并放置到map中
        parseXml();
        System.out.println("parseXml==>"+map);

        //解析注解定义的Bean，并放置到map中
        parseAnno("com.lagou.edu");
        System.out.println("parseAnno==>"+map);

        //对添加了Autowire注解的属性进行注入
        handleAutowiredAndTransactionl();
    }


    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static  Object getBean(String id) {
        return map.get(id);
    }

}
